跳到主要内容

216. 组合总和 III

mid

找出所有相加之和为 nk 个数的组合,且满足下列条件:

  • 只使用数字 1 到 9
  • 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

DFS 回溯

子集问题

k 相当于树的深度,n 是树的宽度

216.组合总和III

func combinationSum3(k int, n int) [][]int {
// k: 层数
var res=make([][]int,0)
var tmp=make([]int,0)
var dfs func(layer, sum, startIndex int)
dfs = func(layer, sum, startIndex int){
if layer==k{
if sum==n{
res=append(res,append([]int{},tmp...))
}
return
}
for i:=startIndex;i<=9;i++{
if sum+i>n{
// 小小剪个枝
continue
}
tmp=append(tmp,i)
dfs(layer+1,sum+i,i+1)
tmp=tmp[:len(tmp)-1]
}
}
dfs(0,0,1)
return res
}

回溯法中递归函数参数很难一次性确定下来,一般先写逻辑,需要啥参数了,填什么参数。